#include <iostream>
using namespace std;
#include <string>
#include <vector>
#include <map>
#include <ctime>

/*
 - 公司今天招聘了10个员工（ABCDEFGHIJ），10名员工进入公司后，需要指派员工在哪个部门工作
 - 员工信息有：姓名， 工资， 部门：策划、美工、研发
 - 随机给10名员工分配部门和工资
 - 通过multimap进行信息的插入 key(部门编号) value(员工)
 - 分部门显示员工信息
*/

#define CEHUA 0
#define MEISHU 1
#define YANFA 2

class Worker
{
public:
    string name;
    int salary;

};

void creatWorker(vector<Worker>& v)
{
    string nameSeed = "ABCDEFGHIJ";
    for (int i = 0;i < 10;i++)
    {
        Worker worker;
        worker.name = "员工";
        worker.name += nameSeed[i];

        worker.salary = rand() % 10000 + 10000;//10000~19999

        //将员工放入到容器中
        v.push_back(worker);
    }
}

// 员工分组
void setGroup(vector<Worker>& v, multimap<int, Worker>& m)
{
    for (vector<Worker>::iterator it = v.begin();it != v.end();it++)
    {
        // 产生随机部门编号
        int deptID = rand() % 3;// 0 1 2

        //将员工插入到分组中
        //key部门编号   value具体员工
        m.insert(make_pair(deptID, *it));
    }
}

void showWorker(multimap<int, Worker>& m)
{
    cout << "策划部门：" << endl;
    multimap<int, Worker>::iterator pos = m.find(CEHUA);

    int count = m.count(CEHUA);//统计策划具体人数
    int index = 0;
    for (;pos != m.end() && index < count;pos++, index++)
    {
        cout << "姓名：" << pos->second.name
            << "  工资：" << pos->second.salary << endl;
    }
    cout << "-------------------------" << endl;

    cout << "美术部门：" << endl;
    multimap<int, Worker>::iterator pos1 = m.find(MEISHU);

    count = m.count(MEISHU);//统计策划具体人数
    index = 0;
    for (;pos1 != m.end() && index < count;pos1++, index++)
    {
        cout << "姓名：" << pos1->second.name
            << "  工资：" << pos1->second.salary << endl;
    }
    cout << "-------------------------" << endl;

    cout << "研发部门：" << endl;
    multimap<int, Worker>::iterator pos2 = m.find(YANFA);

    count = m.count(YANFA);//统计策划具体人数
    index = 0;
    for (;pos2 != m.end() && index < count;pos2++, index++)
    {
        cout << "姓名：" << pos2->second.name
            << "  工资：" << pos2->second.salary << endl;
    }
}

int main()
{
    srand((unsigned int)time(NULL));
    
    //1、创建员工
    vector<Worker>v;
    creatWorker(v);

    //测试
    // for (vector<Worker>::iterator it = v.begin();it != v.end();it++)
    // {
    //     cout << "姓名：" << it->name
    //         << "  工资：" << it->salary << endl;
    // }

    //2、员工分组
    multimap<int, Worker> m;
    setGroup(v, m);

    //3、分组显示员工
    showWorker(m);
    
    return 0;
}